#!/usr/bin/env python
# coding:utf-8

import requests
from hashlib import md5
import os


class Chaojiying_Client(object):
    def __init__(self, username, password, soft_id):
        self.username = username
        password = password.encode('utf8')
        self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection':
            'Keep-Alive',
            'User-Agent':
            'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php',
                          data=params,
                          files=files,
                          headers=self.headers)
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = requests.post(
            'http://upload.chaojiying.net/Upload/ReportError.php',
            data=params,
            headers=self.headers)
        return r.json()


if __name__ == '__main__':
    os.chdir(os.path.abspath(os.path.dirname(__file__)))

    chaojiying = Chaojiying_Client(
        'h521822', 'hj123456',
        '794d92b4ae80aff7b183e1c6c260d553')  #用户中心>>软件ID 生成一个替换 96001
    # im = open('a.jpg', 'rb').read()  #本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
    # print(chaojiying.PostPic(im,
    #                          1902))  #1902 验证码类型  官方网站>>价格体系 3.4+版 print 后要加()

    im = open('1598579694.jpg', 'rb').read()  #本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
    print(chaojiying.PostPic(im,
                             9103))  #1902 验证码类型  官方网站>>价格体系 3.4+版 print 后要加()
